package com.lanbu.system.service.impl;

import org.springframework.stereotype.Service;

import java.util.concurrent.ConcurrentHashMap;

@Service
public class MappingService {

    private final ConcurrentHashMap<String, String> mappingStore = new ConcurrentHashMap<>();

    public void saveMapping(String dialogId, String knowledgeBaseId) {
        mappingStore.put(dialogId, knowledgeBaseId);
    }

    public String getKnowledgeBaseId(String dialogId) {
        return mappingStore.get(dialogId);
    }
}
